# !/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
@作者 杨阳
@时间 2018-06-29
@说明 与session或当前登录用户相关的工具类
@注意事项
"""
import functools

from flask import jsonify
from flask_login import current_user


def Mast_be_login_decorator(func):
    @functools.wraps(func)
    def decorator(*args, **kwargs):
        """
        判断用户是否已经登录的装饰器
        :param func:
        :return:
        """
        # 判断用户名存在且用户名是什么的时候直接那个视图函数
        if current_user.is_authenticated:
            return func(*args, **kwargs)
        else:
            # 如果用户未登录就重定向到登录页面
            return jsonify(success=False, msg="session-timeout")

    return decorator


def Get_user_permission_decorator(func):
    """
    获取用户权限的装饰嚣，但是暂时还没太想好怎么用
    :param func:
    :return:
    """

    def decorator(*args, **kwargs):
        return func(*args, **kwargs)
